/*
 * Copyright 2010 Assert Developments
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package be.ad.coffeebreak.rest.support;

import com.sun.jersey.core.spi.component.ComponentContext;
import com.sun.jersey.core.spi.component.ComponentScope;
import com.sun.jersey.spi.inject.Injectable;
import com.sun.jersey.spi.inject.InjectableProvider;

import javax.ejb.EJB;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.ws.rs.ext.Provider;
import java.lang.reflect.Type;

/**
 * @author Stefan Bangels
 */
@Provider
public class EJBInjectingProvider implements InjectableProvider<EJB, Type> {

    private Context context;

    public EJBInjectingProvider() throws Exception {
        context = new InitialContext();
    }

    public ComponentScope getScope() {
        return ComponentScope.Singleton;
    }

    public Injectable getInjectable(ComponentContext componentContext, final EJB annotation, final Type type) {
        return new Injectable<Object>() {
            @Override
            @SuppressWarnings("unchecked")
            public Object getValue() {
                try {
                    return context.lookup(annotation.mappedName());
                } catch (NamingException e) {
                    throw new RuntimeException(e);
                }
            }
        };
    }

}